/*
 * cpu_use.c
 *
 *  Created on: 2022.3.3
 *      Author: huan
 */

#include <rtthread.h>
#include <rthw.h>
#include "head_user.h"

#define CPU_USAGE_CALC_TICK    100
#define CPU_USAGE_LOOP         1000

static rt_uint8_t  cpu_usage_major, cpu_usage_minor;
static rt_uint32_t total_count;

static void cpu_usage_idle_hook(void)
{
    rt_tick_t tick;
    rt_uint32_t count = 0;
    volatile rt_uint32_t loop;

    if (total_count == 0)
    {
        /* get total count */
        rt_enter_critical();

        tick = rt_tick_get();
        while(rt_tick_get() - tick < (CPU_USAGE_CALC_TICK + 1))
        {
            total_count ++;
            loop = 0;
            while (loop < CPU_USAGE_LOOP) loop++;
        }

        rt_exit_critical();
    }

    /* get CPU usage */
    tick = rt_tick_get();
    while (rt_tick_get() - tick < CPU_USAGE_CALC_TICK)
    {
        count ++;
        loop  = 0;
        while (loop < CPU_USAGE_LOOP) loop ++;
    }
    /* calculate major and minor */
    if (count < total_count)
    {
        count = total_count - count;
        cpu_usage_major = (count * 100) / total_count;
        cpu_usage_minor = ((count * 100) % total_count) * 100 / total_count;
    }
    else
    {
        total_count = count;
        /* no CPU usage */
        cpu_usage_major = 0;
        cpu_usage_minor = 0;
    }
}

void cpu_usage_get(rt_uint8_t *major, rt_uint8_t *minor)
{
    RT_ASSERT(major != RT_NULL);
    RT_ASSERT(minor != RT_NULL);

    *major = cpu_usage_major;
    *minor = cpu_usage_minor;
}

static int cpu_usage_init(void)
{
    /* set idle thread hook */
    rt_thread_idle_sethook(cpu_usage_idle_hook);
    return 0;
}

INIT_BOARD_EXPORT(cpu_usage_init);
